home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / lib / python2.5 / pprint.pyc (.txt) < prev    next >
Python Compiled Bytecode  |  2008-10-29  |  10KB  |  354 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.5)
  3.  
  4. """Support to pretty-print lists, tuples, & dictionaries recursively.
  5.  
  6. Very simple, but useful, especially in debugging data structures.
  7.  
  8. Classes
  9. -------
  10.  
  11. PrettyPrinter()
  12.     Handle pretty-printing operations onto a stream using a configured
  13.     set of formatting parameters.
  14.  
  15. Functions
  16. ---------
  17.  
  18. pformat()
  19.     Format a Python object into a pretty-printed representation.
  20.  
  21. pprint()
  22.     Pretty-print a Python object to a stream [default is sys.stdout].
  23.  
  24. saferepr()
  25.     Generate a 'standard' repr()-like value, but protect against recursive
  26.     data structures.
  27.  
  28. """
  29. import sys as _sys
  30. from cStringIO import StringIO as _StringIO
  31. __all__ = [
  32.     'pprint',
  33.     'pformat',
  34.     'isreadable',
  35.     'isrecursive',
  36.     'saferepr',
  37.     'PrettyPrinter']
  38. _commajoin = ', '.join
  39. _id = id
  40. _len = len
  41. _type = type
  42.  
  43. def pprint(object, stream = None, indent = 1, width = 80, depth = None):
  44.     '''Pretty-print a Python object to a stream [default is sys.stdout].'''
  45.     printer = PrettyPrinter(stream = stream, indent = indent, width = width, depth = depth)
  46.     printer.pprint(object)
  47.  
  48.  
  49. def pformat(object, indent = 1, width = 80, depth = None):
  50.     '''Format a Python object into a pretty-printed representation.'''
  51.     return PrettyPrinter(indent = indent, width = width, depth = depth).pformat(object)
  52.  
  53.  
  54. def saferepr(object):
  55.     '''Version of repr() which can handle recursive data structures.'''
  56.     return _safe_repr(object, { }, None, 0)[0]
  57.  
  58.  
  59. def isreadable(object):
  60.     '''Determine if saferepr(object) is readable by eval().'''
  61.     return _safe_repr(object, { }, None, 0)[1]
  62.  
  63.  
  64. def isrecursive(object):
  65.     '''Determine if object requires a recursive representation.'''
  66.     return _safe_repr(object, { }, None, 0)[2]
  67.  
  68.  
  69. class PrettyPrinter:
  70.     
  71.     def __init__(self, indent = 1, width = 80, depth = None, stream = None):
  72.         '''Handle pretty printing operations onto a stream using a set of
  73.         configured parameters.
  74.  
  75.         indent
  76.             Number of spaces to indent for each level of nesting.
  77.  
  78.         width
  79.             Attempted maximum number of columns in the output.
  80.  
  81.         depth
  82.             The maximum depth to print out nested structures.
  83.  
  84.         stream
  85.             The desired output stream.  If omitted (or false), the standard
  86.             output stream available at construction will be used.
  87.  
  88.         '''
  89.         indent = int(indent)
  90.         width = int(width)
  91.         if not indent >= 0:
  92.             raise AssertionError, 'indent must be >= 0'
  93.         if not depth is None and depth > 0:
  94.             raise AssertionError, 'depth must be > 0'
  95.         if not width:
  96.             raise AssertionError, 'width must be != 0'
  97.         self._depth = depth
  98.         self._indent_per_level = indent
  99.         self._width = width
  100.         if stream is not None:
  101.             self._stream = stream
  102.         else:
  103.             self._stream = _sys.stdout
  104.  
  105.     
  106.     def pprint(self, object):
  107.         self._format(object, self._stream, 0, 0, { }, 0)
  108.         self._stream.write('\n')
  109.  
  110.     
  111.     def pformat(self, object):
  112.         sio = _StringIO()
  113.         self._format(object, sio, 0, 0, { }, 0)
  114.         return sio.getvalue()
  115.  
  116.     
  117.     def isrecursive(self, object):
  118.         return self.format(object, { }, 0, 0)[2]
  119.  
  120.     
  121.     def isreadable(self, object):
  122.         (s, readable, recursive) = self.format(object, { }, 0, 0)
  123.         if readable:
  124.             pass
  125.         return not recursive
  126.  
  127.     
  128.     def _format(self, object, stream, indent, allowance, context, level):
  129.         level = level + 1
  130.         objid = _id(object)
  131.         if objid in context:
  132.             stream.write(_recursion(object))
  133.             self._recursive = True
  134.             self._readable = False
  135.             return None
  136.         
  137.         rep = self._repr(object, context, level - 1)
  138.         typ = _type(object)
  139.         sepLines = _len(rep) > self._width - 1 - indent - allowance
  140.         write = stream.write
  141.         if sepLines:
  142.             r = getattr(typ, '__repr__', None)
  143.             if issubclass(typ, dict) and r is dict.__repr__:
  144.                 write('{')
  145.                 if self._indent_per_level > 1:
  146.                     write((self._indent_per_level - 1) * ' ')
  147.                 
  148.                 length = _len(object)
  149.                 if length:
  150.                     context[objid] = 1
  151.                     indent = indent + self._indent_per_level
  152.                     items = object.items()
  153.                     items.sort()
  154.                     (key, ent) = items[0]
  155.                     rep = self._repr(key, context, level)
  156.                     write(rep)
  157.                     write(': ')
  158.                     self._format(ent, stream, indent + _len(rep) + 2, allowance + 1, context, level)
  159.                     if length > 1:
  160.                         for key, ent in items[1:]:
  161.                             rep = self._repr(key, context, level)
  162.                             write(',\n%s%s: ' % (' ' * indent, rep))
  163.                             self._format(ent, stream, indent + _len(rep) + 2, allowance + 1, context, level)
  164.                         
  165.                     
  166.                     indent = indent - self._indent_per_level
  167.                     del context[objid]
  168.                 
  169.                 write('}')
  170.                 return None
  171.             
  172.             if (issubclass(typ, list) or r is list.__repr__ or issubclass(typ, tuple)) and r is tuple.__repr__:
  173.                 if issubclass(typ, list):
  174.                     write('[')
  175.                     endchar = ']'
  176.                 else:
  177.                     write('(')
  178.                     endchar = ')'
  179.                 if self._indent_per_level > 1:
  180.                     write((self._indent_per_level - 1) * ' ')
  181.                 
  182.                 length = _len(object)
  183.                 if length:
  184.                     context[objid] = 1
  185.                     indent = indent + self._indent_per_level
  186.                     self._format(object[0], stream, indent, allowance + 1, context, level)
  187.                     if length > 1:
  188.                         for ent in object[1:]:
  189.                             write(',\n' + ' ' * indent)
  190.                             self._format(ent, stream, indent, allowance + 1, context, level)
  191.                         
  192.                     
  193.                     indent = indent - self._indent_per_level
  194.                     del context[objid]
  195.                 
  196.                 if issubclass(typ, tuple) and length == 1:
  197.                     write(',')
  198.                 
  199.                 write(endchar)
  200.                 return None
  201.             
  202.         
  203.         write(rep)
  204.  
  205.     
  206.     def _repr(self, object, context, level):
  207.         (repr, readable, recursive) = self.format(object, context.copy(), self._depth, level)
  208.         if not readable:
  209.             self._readable = False
  210.         
  211.         if recursive:
  212.             self._recursive = True
  213.         
  214.         return repr
  215.  
  216.     
  217.     def format(self, object, context, maxlevels, level):
  218.         """Format object for a specific context, returning a string
  219.         and flags indicating whether the representation is 'readable'
  220.         and whether the object represents a recursive construct.
  221.         """
  222.         return _safe_repr(object, context, maxlevels, level)
  223.  
  224.  
  225.  
  226. def _safe_repr(object, context, maxlevels, level):
  227.     typ = _type(object)
  228.     if typ is str:
  229.         if 'locale' not in _sys.modules:
  230.             return (repr(object), True, False)
  231.         
  232.         if "'" in object and '"' not in object:
  233.             closure = '"'
  234.             quotes = {
  235.                 '"': '\\"' }
  236.         else:
  237.             closure = "'"
  238.             quotes = {
  239.                 "'": "\\'" }
  240.         qget = quotes.get
  241.         sio = _StringIO()
  242.         write = sio.write
  243.         for char in object:
  244.             if char.isalpha():
  245.                 write(char)
  246.                 continue
  247.             write(qget(char, repr(char)[1:-1]))
  248.         
  249.         return ('%s%s%s' % (closure, sio.getvalue(), closure), True, False)
  250.     
  251.     r = getattr(typ, '__repr__', None)
  252.     if issubclass(typ, dict) and r is dict.__repr__:
  253.         if not object:
  254.             return ('{}', True, False)
  255.         
  256.         objid = _id(object)
  257.         if maxlevels and level > maxlevels:
  258.             return ('{...}', False, objid in context)
  259.         
  260.         if objid in context:
  261.             return (_recursion(object), False, True)
  262.         
  263.         context[objid] = 1
  264.         readable = True
  265.         recursive = False
  266.         components = []
  267.         append = components.append
  268.         level += 1
  269.         saferepr = _safe_repr
  270.         for k, v in sorted(object.items()):
  271.             (krepr, kreadable, krecur) = saferepr(k, context, maxlevels, level)
  272.             (vrepr, vreadable, vrecur) = saferepr(v, context, maxlevels, level)
  273.             append('%s: %s' % (krepr, vrepr))
  274.             if readable and kreadable:
  275.                 pass
  276.             readable = vreadable
  277.             if krecur or vrecur:
  278.                 recursive = True
  279.                 continue
  280.         
  281.         del context[objid]
  282.         return ('{%s}' % _commajoin(components), readable, recursive)
  283.     
  284.     if (issubclass(typ, list) or r is list.__repr__ or issubclass(typ, tuple)) and r is tuple.__repr__:
  285.         if issubclass(typ, list):
  286.             if not object:
  287.                 return ('[]', True, False)
  288.             
  289.             format = '[%s]'
  290.         elif _len(object) == 1:
  291.             format = '(%s,)'
  292.         elif not object:
  293.             return ('()', True, False)
  294.         
  295.         format = '(%s)'
  296.         objid = _id(object)
  297.         if maxlevels and level > maxlevels:
  298.             return (format % '...', False, objid in context)
  299.         
  300.         if objid in context:
  301.             return (_recursion(object), False, True)
  302.         
  303.         context[objid] = 1
  304.         readable = True
  305.         recursive = False
  306.         components = []
  307.         append = components.append
  308.         level += 1
  309.         for o in object:
  310.             (orepr, oreadable, orecur) = _safe_repr(o, context, maxlevels, level)
  311.             append(orepr)
  312.             if not oreadable:
  313.                 readable = False
  314.             
  315.             if orecur:
  316.                 recursive = True
  317.                 continue
  318.         
  319.         del context[objid]
  320.         return (format % _commajoin(components), readable, recursive)
  321.     
  322.     rep = repr(object)
  323.     if rep:
  324.         pass
  325.     return (rep, not rep.startswith('<'), False)
  326.  
  327.  
  328. def _recursion(object):
  329.     return '<Recursion on %s with id=%s>' % (_type(object).__name__, _id(object))
  330.  
  331.  
  332. def _perfcheck(object = None):
  333.     import time as time
  334.     if object is None:
  335.         object = [
  336.             ('string', (1, 2), [
  337.                 3,
  338.                 4], {
  339.                 5: 6,
  340.                 7: 8 })] * 100000
  341.     
  342.     p = PrettyPrinter()
  343.     t1 = time.time()
  344.     _safe_repr(object, { }, None, 0)
  345.     t2 = time.time()
  346.     p.pformat(object)
  347.     t3 = time.time()
  348.     print '_safe_repr:', t2 - t1
  349.     print 'pformat:', t3 - t2
  350.  
  351. if __name__ == '__main__':
  352.     _perfcheck()
  353.  
  354.